package com.heima.wemedia.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.heima.aliyun.scan.GreenScan;
import com.heima.aliyun.scan.ScanResult;
import com.heima.common.exception.CustException;
import com.heima.common.message.PublishArticleConstants;
import com.heima.common.wemedia.WemediaConstants;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import com.heima.wemedia.service.WmNewsService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.AmqpException;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {
    @Autowired
    AdminFeign adminFeign;
    @Autowired
    GreenScan greenScan;
    @Autowired
    WmNewsService wmNewsService;
    @Value("${file.oss.web-site}")
    String webSite;
    @Autowired
    RabbitTemplate rabbitTemplate;

    @Override
    public void autoScanWmNews(Integer id) {
        log.info("成功接收到文章自动审核消息  待审核的文章id -> {}",id);
        // 1. 根据id查询自媒体文章数据
        WmNews wmNews = wmNewsService.getById(id);
        if (wmNews==null){
            log.info("文章不存在，自动审核失败");
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST,"文章不存在，自动审核失败");
        }
        // 2. 判断文章的状态是否为 1  (防止重复消费)
        if (!WemediaConstants.WM_NEWS_SUMMIT_STATUS.equals(wmNews.getStatus())){
            log.info("文章不是待审核状态，自动审核失败  文章id -> {}",id);
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST,"文章不是待审核状态，自动审核失败");
        }
        // 3. 开始审核
        // 3.0  抽取出 文章的所有文本内容  和  所有的图片路径
        Map contentAndImages = parseContentAndImages(wmNews);
        // 3.1  DFA自管理的敏感词审核
        boolean sensitiveScan = checkSensitive((String)contentAndImages.get("content"),wmNews);
        if (!sensitiveScan){
        // 失败  状态改为2   设置失败原因:
            log.info("文章审核未通过，文章含有敏感词信息  文章Id -> {}",id);
            return;
        }
        // 通过  继续下一步
        // 3.2  阿里云内容安全文本审核
        boolean aliyunTextScan =aliyunCheckText((String)contentAndImages.get("content"),wmNews);
        if (!aliyunTextScan){
            // 失败
            log.info("阿里云文本审核未通过，文章含有违规信息  文章Id -> {}",id);
            return;
        }

        // 3.3  阿里云内容安全图片审核
        List<String> images = (List<String>) contentAndImages.get("images");
        if (!CollectionUtils.isEmpty(images)) {
            boolean aliyunImageScan =aliyunCheckImage(images,wmNews);
            if (!aliyunImageScan){
                log.info("文章阿里云图片审核未通过，文章图片中有违规信息 文章id: {}",id);
                return;
            }
        }
        // 4.  如果全部审核通过，将文章状态改为8 审核通过
        updateWmNews(wmNews,WmNews.Status.SUCCESS.getCode(),"审核通过");

        // 5. TODO 根据文章的具体发布时间   发送延时发布消息
        long publishTime = wmNews.getPublishTime().getTime();
        long nowTime = System.currentTimeMillis();
        long remainTime = publishTime - nowTime;
        //发布文章
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE
                , PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY
                , wmNews.getId(), new MessagePostProcessor() {
                    /**
                     * 消息的后置处理器，可以对消息属性进行设置
                     * @param message
                     * @return
                     * @throws AmqpException
                     */
                    @Override
                    public Message postProcessMessage(Message message) throws AmqpException {
                        // 设置 x-delay 延迟时间
                        message.getMessageProperties().setHeader("x-delay",remainTime<=0?0:remainTime);
                        return message;
                    }
                });
        log.info("成功发送延时发布文章消息   文章id:{}  延迟时间: {}",wmNews.getId(),remainTime);
    }

    /**
     * 阿里云内容安全图片审核
     * @param images
     * @param wmNews
     * @return
     */
    private boolean aliyunCheckImage(List images, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.imageUrlScan(images);
            //2.  判断suggestion   block 2  review 3  pass 返回true    null  3
            String suggestion = scanResult.getSuggestion();
            //3   根据 suggestion的值修改文章状态
            switch (suggestion){
                // 失败  状态改为2   设置失败原因:
                case "block":
                    updateWmNews(wmNews,WmNews.Status.FAIL.getCode(),"文章图片中有违规信息, "+scanResult.getLabel());
                    break;
                // 待人工审核  状态改为3
                case "review":
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"文章中有不确定的图片信息,转为人工审核 ");
                    break;
                // 通过  继续下一步
                case "pass":
                    flag = true;
                    break;
                default:
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"阿里云 图片审核出现异常， 当前文章转为人工审核状态");
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 如果阿里云调用异常，将状态修改为3  打印错误日志信息
            log.error("阿里云 图片审核出现异常， 当前文章转为人工审核状态 ");
            updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"阿里云 图片审核出现异常， 当前文章转为人工审核状态");
        }
        return flag;
    }

    /**
     * 阿里云内容安全文本审核
     * @param content
     * @param wmNews
     * @return
     */
    private boolean aliyunCheckText(String content, WmNews wmNews) {
        boolean flag=false;
        try {
            ScanResult scanResult = greenScan.greenTextScan(content);
            //2.  判断suggestion   block 2  review 3  pass 返回true    null  3
            String suggestion = scanResult.getSuggestion();
            //3   根据 suggestion的值修改文章状态
            switch (suggestion){
                // 失败  状态改为2   设置失败原因:
                case "block":
                    updateWmNews(wmNews,WmNews.Status.FAIL.getCode(),"文章内容中有违规信息, "+scanResult.getLabel());
                    break;
                // 待人工审核  状态改为3
                case "review":
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"文章内容中有不确定的信息,转为人工审核 ");
                    break;
                // 通过  继续下一步
                case "pass":
                    flag = true;
                    break;
                default:
                    updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"阿里云 文本审核出现异常， 当前文章转为人工审核状态");
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            log.info("阿里云文本审核调用异常 当前文章转为人工审核状态");
            updateWmNews(wmNews,WmNews.Status.ADMIN_AUTH.getCode(),"阿里云文本审核调用异常 当前文章转为人工审核状态" );
        }
        return flag;
    }


    /**
     * DFA自管理的敏感词审核
     * @param content
     * @param wmNews
     * @return
     */
    private boolean checkSensitive(String content, WmNews wmNews) {
        ResponseResult<List<String>> sensitivesResult = adminFeign.sensitives();
        if (!sensitivesResult.checkCode()) {
            CustException.cust(AppHttpCodeEnum.REMOTE_SERVER_ERROR,"查询敏感词列表，远程调用失败");
        }
        List<String> sensitives = sensitivesResult.getData();
        // TODO 先从redis中 查询是否有敏感词列表, 如果redis中没有 远程调用数据库查询
        // TODO 如果有直接获取redis中的缓存
        SensitiveWordUtil.initMap(sensitives);
        Map<String, Integer> map = SensitiveWordUtil.matchWords(content);
        if (CollectionUtils.isNotEmpty(map)){
            updateWmNews(wmNews,WmNews.Status.FAIL.getCode(), "文章内容中含有敏感词 "+map);
            return false;
        }

            return true;
    }

    private void updateWmNews(WmNews wmNews, Short status,String reason) {
        wmNews.setStatus(status);
        wmNews.setReason(reason);
        wmNewsService.updateById(wmNews);
    }

    /**
     * 抽取文章内容 和  图片路径集合的方法
     * @param wmNews
     * @return
     */
    private Map parseContentAndImages(WmNews wmNews) {
        String content = wmNews.getContent();//文章内容
        //文章所有文字
        List<Map> contentMapList = JSON.parseArray(content, Map.class);
        String allContent = contentMapList.stream()
                .filter(m -> "text".equals(m.get("type")))
                .map(m -> String.valueOf(m.get("value")))
                .collect(Collectors.joining("----"));
        //文章所有文字加上标题
        allContent += "-----" + wmNews.getTitle();
        //抽取所有文章图片路径
        List<String> allImage = contentMapList.stream()
                .filter(m -> "image".equals(m.get("type")))
                .map(m -> String.valueOf(m.get("value")))
                .collect(Collectors.toList());
        String coverImages = wmNews.getImages();
        if (StringUtils.isNotBlank(coverImages)) {
            // material/2022/7/20220728/3787d5ef88ad4e8cbbbb30ddeb35b1e80001.jpg
            // webSite + 图片相对路径.jpg    ,    图片相对路径.jpg,     图片相对路径.jpg
            // 1. 按照逗号切割  得到数组
            // 2. 基于数组 获取流对象
            // 3. 使用map映射方法，将每个路径添加访问前缀
            // 4. 收集到一个集合中
            List<String> coverList = Arrays.stream(coverImages.split(","))
                    .map(url -> webSite + url)
                    .collect(Collectors.toList());
            // 将内容图片路径 和 封面图片路径合到一起
            allImage.addAll(coverList);
        }

        allImage = allImage.stream().distinct().collect(Collectors.toList());

        Map result = new HashMap();
        result.put("content",allContent);
        result.put("images",allImage);
        return result;
    }
}
